#pragma once
#include <iostream>
#include <vector>
#include <assert.h>

namespace log
{
#define DEFAULT_BUFFER_SIZE 1 * 1024 * 1024   // 缓冲区默认存储大小
#define THRESHOLD_BUFFER_SIZE 8 * 1024 * 1024 // 阈值
#define INCREMENT_BUFFER_SIZE 1 * 1024 * 1024 // 达到阈值后的增量

    class Buffer
    {
    public:
        Buffer() : _buffer(DEFAULT_BUFFER_SIZE), _read_pos(0), _write_pos(0) {}
        // 向缓冲区写入数据
        void push(const char *data, int len)
        {
            //不关注是扩容还是阻塞,由looper来关注
            Expansion(len);
            std::copy(data, data + len, &_buffer[_write_pos]); //将data拷贝到缓冲区
            moveWrite(len); //写指针向后偏移
        }
        // 返回可读数据的起始位置
        const char *begin()
        {
            return &_buffer[_read_pos];
        }
        // 返回可写数据的长度(仅针对"1.不处理,阻塞/直接返回"这种情况)
        int writeAbleLen()
        {
            return (_buffer.size() - _write_pos);
        }
        // 返回可读数据的长度
        int readAbleLen()
        {
            return (_write_pos - _read_pos);
        }
        // 对读指针进行向后偏移操作
        void moveRead(int len)
        {
            assert(len <= readAbleLen());
            _read_pos += len;
        }
        // 重置读写位置,初始化缓冲区
        void reset()
        {
            _read_pos = 0;
            _write_pos = 0;
        }
        // 对缓冲区进行交换
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_read_pos, buffer._read_pos);
            std::swap(_write_pos, buffer._write_pos);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return _read_pos == _write_pos;
        }

    private:
        //扩容
        void Expansion(int len)
        {
            //如果可写空间还足够,那么不需要扩容
            if(len <= writeAbleLen())
            {
                return;
            }
            //反之,扩容
            else
            {
                int new_size = 0;
                //小于阈值,翻倍增加
                if(_buffer.size() < THRESHOLD_BUFFER_SIZE)
                {
                    new_size = _buffer.size() * 2 + len; //扩容2倍后在加上len,是为了预防即使扩容2倍,依然小于len的情况,预防len的大小过大,下同
                }
                //反之,固定增加
                else
                {
                    new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len;
                }
                _buffer.resize(new_size);
            }
        }
        // 对写指针进行向后偏移操作
        void moveWrite(int len)
        {
            assert(_write_pos + len <= _buffer.size());
            _write_pos += len;
        }

    private:
        std::vector<char> _buffer; // 缓冲区
        int _read_pos;             // 读下标
        int _write_pos;            // 写下标
    };
};